package com.hulei.studyproject.nacos.listener;

import com.alibaba.cloud.commons.lang.StringUtils;
import com.alibaba.fastjson.JSONArray;
import lombok.extern.slf4j.Slf4j;
import net.coobird.thumbnailator.Thumbnails;
import org.csource.common.MyException;
import org.csource.common.NameValuePair;
import org.csource.fastdfs.*;
import org.springframework.stereotype.Component;
import org.springframework.web.multipart.MultipartFile;

import javax.imageio.ImageIO;
import javax.imageio.stream.ImageOutputStream;
import java.awt.image.BufferedImage;
import java.io.*;
import java.util.Objects;
import java.util.Properties;

/**
 * 分布式文件系统 Java 客户端工具类 具体功能：文件上传、下载、替换、删除、查询文件元数据、查看文件详情
 * 继承nacos监听器，实现nacos配置读取和实时监听配置修改
 * @Title: FastDFSClientUtil
 * @Description: TODO
 * @author: hulei
 * @date: 2023/8/8 9:10
 * @Version: 1.0
 */
@Component
@Slf4j
public class FastDFSClientUtil extends NacosJsonDefaultListener<JSONArray> {
    /**
     * Storage 存储服务器客户端
     */
    private static StorageClient storageClient = null;

    private static final Properties props = new Properties();

    /**
     * 文件上传
     *
     * @param inputStream 上传文件 字节输入流
     * @param fileName    上传文件的原始名
     * @return String[]
     */
    public static String[] uploadFile(InputStream inputStream, String fileName) {
        try {
            // 准备字节数组
            byte[] fileBuff = null;
            // 文件元数据
            if (inputStream != null) {
                // 查看文件的长度
                int len = inputStream.available();
                // 创建对应长度的字节数组
                fileBuff = new byte[len];
                // 将输入流中的字节内容，读到字节数组中
                inputStream.read(fileBuff);
            }
            log.info("文件上传，加载配置文件");
            init();
            // 上传文件。 参数含义：要上传的文件的内容（使用字节数组传递），上传的文件的类型（扩展名），元数据
            log.info("上传文件");
            return storageClient.upload_file(fileBuff, getFileExt(fileName), null);
        } catch (IOException e) {
            log.error("上传文件失败发生IO异常:{}",e.getMessage());
        } catch (MyException e) {
            log.error("上传文件自定义异常:{}",e.getMessage());
        }
        return null;
    }

    /**
     * 文件上传
     *
     * @param file     上传文件 File类型
     * @param fileName 上传的文件的原始名
     * @return String[]
     */
    public static String[] uploadFile(File file, String fileName) {
        try (FileInputStream fis = new FileInputStream(file)) {
            return uploadFile(fis, fileName);
        } catch (FileNotFoundException e) {
            log.error("上传文件失败发生FileNotFoundException异常:{}",e.getMessage());
        } catch (IOException e) {
            log.error("上传文件失败发生IO异常:{}",e.getMessage());
        }
        return null;
    }

    /**
     * 文件上传
     *
     * @param file     上传文件MultipartFile类型
     * @param fileName 上传文件的原始名
     * @return String[]
     */
    public static String[] uploadFile(MultipartFile file, String fileName) {
        try {
            log.info("判断图片大小：" + file.getSize());
            if (file.getSize() <= 0) return null;
            log.info("转换为File对象");
            InputStream ins = file.getInputStream();
            File f = new File(Objects.requireNonNull(file.getOriginalFilename()));
            OutputStream os = new FileOutputStream(f);
            int bytesRead;
            byte[] buffer = new byte[8192];
            while ((bytesRead = ins.read(buffer, 0, 8192)) != -1) {
                os.write(buffer, 0, bytesRead);
            }
            os.close();
            ins.close();
            String[] fileIds = uploadFile(f, fileName);
            log.info("返回文件ID数组");
            if (f.exists())f.delete();
            return fileIds;
        } catch (FileNotFoundException e) {
            log.error("上传文件失败发生FileNotFoundException异常:{}",e.getMessage());
        } catch (IOException e) {
            log.error("上传文件失败发生IO异常:{}",e.getMessage());
        }
        return null;
    }

    /**
     * 图片缩略图上传
     *
     * @param file 上传的文件
     * @return String[]
     */
    public static String[] uploadFileThumbnail(MultipartFile file) {
        try {
            String fileName = file.getOriginalFilename();
            // 获取图片文件格式
            String fileTypeName = getFileExt(fileName);
            File f;
            // 缩略图文件流处理
            BufferedImage bufferedImage = Thumbnails.of(file.getInputStream()).size(150, 150).asBufferedImage();
            ByteArrayOutputStream os = new ByteArrayOutputStream();
            ImageOutputStream imOut = ImageIO.createImageOutputStream(os);
            ImageIO.write(bufferedImage, fileTypeName, imOut);
            // 上传缩略图
            if (file.getSize() > 0) {
                InputStream ins = new ByteArrayInputStream(os.toByteArray());
                f = new File(Objects.requireNonNull(file.getOriginalFilename()));
                OutputStream outputStream = new FileOutputStream(f);
                int bytesRead;
                byte[] buffer = new byte[8192];
                while ((bytesRead = ins.read(buffer, 0, 8192)) != -1) {
                    outputStream.write(buffer, 0, bytesRead);
                }
                outputStream.close();
                ins.close();
                String[] fileIds = uploadFile(f, fileName);
                if (f.exists()) f.delete();
                return fileIds;
            }
            return null;
        } catch (IOException e) {
            log.error("上传文件失败发生IO异常:{}",e.getMessage());
        }
        return null;
    }

    /**
     * 获取文件后缀名（不带点）
     *
     * @param fileName 文件名称
     * @return 如："jpg" or ""
     */
    public static String getFileExt(String fileName) {
        if (StringUtils.isBlank(fileName) || !fileName.contains(".")) {
            return "";
        }
        // 不带最后的点
        return fileName.substring(fileName.lastIndexOf(".") + 1);
    }

    /**
     * 获取文件详情
     *
     * @param groupName      组/卷名，默认值：group1
     * @param remoteFileName 文件名，例如："M00/00/00/wKgKZl9tkTAJAanAADhaCZ_RF0495.jpg"
     * @return 文件详情
     */
    public static FileInfo getFileInfo(String groupName, String remoteFileName) {
        try {
            return storageClient.get_file_info(groupName == null ? "group1" : groupName, remoteFileName);
        } catch (IOException e) {
            log.error("上传文件失败发生IO异常:{}",e.getMessage());
        } catch (MyException e) {
            log.error("上传文件自定义异常:{}",e.getMessage());
        }
        return null;
    }

    /**
     * 获取元数据
     *
     * @param groupName      组/卷名，默认值：group1
     * @param remoteFileName 文件名，例如："M00/00/00/wKgKZl9tkTAJAanAADhaCZ_RF0495.jpg"
     * @return 文件的元数据数组
     */
    public static NameValuePair[] getMetaData(String groupName, String remoteFileName) {
        try {
            // 根据组名和文件名通过 Storage 客户端获取文件的元数据数组
            return storageClient.get_metadata(groupName == null ? "group1" : groupName, remoteFileName);
        } catch (IOException e) {
            log.error("上传文件失败发生IO异常:{}",e.getMessage());
        } catch (MyException e) {
            log.error("上传文件自定义异常:{}",e.getMessage());
        }
        return null;
    }

    /**
     * 文件下载
     *
     * @param groupName      组/卷名，默认值：group1
     * @param remoteFileName 文件名，例如："M00/00/00/wKgKZl9tkTAJAanAADhaCZ_RF0495.jpg"
     * @return 文件的字节输入流
     */
    public static InputStream downloadFile(String groupName, String remoteFileName) {
        try {
            init();
            // 根据组名和文件名通过 Storage 客户端获取文件的字节数组
            byte[] bytes = storageClient.download_file(groupName == null ? "group1" : groupName, remoteFileName);
            // 返回字节流对象
            return new ByteArrayInputStream(bytes);
        } catch (IOException e) {
            log.error("上传文件失败发生IO异常:{}",e.getMessage());
        } catch (MyException e) {
            log.error("上传文件自定义异常:{}",e.getMessage());
        }
        return null;
    }

    /**
     * 文件下载
     *
     * @param groupName      组/卷名，默认值：group1
     * @param remoteFileName 文件名，例如："M00/00/00/wKgKZl9tkTAJAanAADhaCZ_RF0495.jpg"
     * @return 文件的字节数组
     */
    public static byte[] downloadFileByte(String groupName, String remoteFileName) {
        try {
            init();
            // 根据组名和文件名通过 Storage 客户端获取文件的字节数组
            return storageClient.download_file(groupName == null ? "group1" : groupName, remoteFileName);
        } catch (IOException e) {
            log.error("上传文件失败发生IO异常:{}",e.getMessage());
        } catch (MyException e) {
            log.error("上传文件自定义异常:{}",e.getMessage());
        }
        return null;
    }

    /**
     * 文件删除
     *
     * @param groupName      组/卷名，默认值：group1
     * @param remoteFileName 文件名，例如："M00/00/00/wKgKZl9tkTAJAanAADhaCZ_RF0495.jpg"
     * @return 0为成功，非0为失败
     */
    public static int deleteFile(String groupName, String remoteFileName) {
        int result = -1;
        try {
            init();
            // 根据组名和文件名通过 Storage 客户端删除文件
            result = storageClient.delete_file(groupName == null ? "group1" : groupName, remoteFileName);
        } catch (IOException e) {
            log.error("上传文件失败发生IO异常:{}",e.getMessage());
        } catch (MyException e) {
            log.error("上传文件自定义异常:{}",e.getMessage());
        }
        return result;
    }

    /**
     * 修改一个已经存在的文件
     *
     * @param oldGroupName 旧组名
     * @param oldFileName  旧文件名
     * @param file         新文件
     * @param fileName     新文件名
     * @return String[]
     */
    public static String[] modifyFile(String oldGroupName, String oldFileName, File file, String fileName) {
        // 先上传
        String[] fileIds = uploadFile(file, fileName);
        if (fileIds == null) {
            return null;
        }
        // 再删除
        int delResult = deleteFile(oldGroupName, oldFileName);
        if (delResult != 0) {
            return null;
        }
        return fileIds;
    }

    @Override
    protected Class<? extends JSONArray> JsonToObjectClass() {
        return JSONArray.class;
    }

    @Override
    protected String getDataId() {
        return "fdfs-client.properties";
    }

    @Override
    protected void compile(String dataStr) {
        String[] configs = dataStr.split("\n");
        for (String config : configs) {
            String[] rows = config.split("=");
            props.setProperty(rows[0].trim(), rows[1].trim());
        }
        init();
    }

    /**
     * 文件服务配置初始化
     */
    public static void init() {
        try {
            // 加载配置文件
            ClientGlobal.initByProperties(props);
            // 初始化 Tracker 客户端
            TrackerClient trackerClient = new TrackerClient(ClientGlobal.g_tracker_group);
            // 初始化 Tracker 服务端
            TrackerServer trackerServer = trackerClient.getConnection();
            // 初始化 Storage 服务端
            StorageServer storageServer = trackerClient.getStoreStorage(trackerServer);
            // 初始化 Storage 客户端
            storageClient = new StorageClient(trackerServer, storageServer);
        } catch (IOException e) {
            log.error("上传文件失败发生IO异常:{}",e.getMessage());
        } catch (MyException e) {
            log.error("上传文件自定义异常:{}",e.getMessage());
        }
    }
}

